<template xlang="wxml">
	<view class="tki-tree">
		<view class="tki-tree-mask" :class="{'show':showTree}" @tap="_cancel"></view>
		<view class="tki-tree-cnt" :class="{'show':showTree}">
			<view class="tki-tree-bar">
				<view class="tki-tree-bar-cancel" :style="{'color':cancelColor}" hover-class="hover-c" @tap="_cancel">取消
				</view>
				<view class="tki-tree-bar-title" :style="{'color':titleColor}">{{title}}</view>
				<view class="tki-tree-bar-confirm" :style="{'color':confirmColor}" hover-class="hover-c"
					@tap="_confirm">确定</view>
			</view>
			<view class="tki-tree-view">
				<view class="tki-tree-view-st">
					<!-- <uni-search-bar :radius="100" @confirm="search" @clear="search" :cancelButton="false"></uni-search-bar> -->
					<u-search placeholder="请输入名称" v-model="keyword" @search="search" @clear="search"
						@custom="search" style="width:100%"></u-search>
				</view>
				<scroll-view class="tki-tree-view-sc" :scroll-y="true">
					<block v-for="(item, index) in treeList" :key="index">
						<view class="tki-tree-item" :style="[{
							paddingLeft: item.rank*15 + 'px',
							zIndex: item.rank*-1 +50
						}]" :class="{
							border: border === true,
							show: item.show,
							last: item.lastRank,
							showchild: item.showChild,
							open: item.open,
						}">
							<view class="tki-tree-label" @tap.stop="_treeItemTap(item, index)">
								<image class="tki-tree-icon"
									:src="item.lastRank ? lastIcon : item.showChild ? currentIcon : defaultIcon">
								</image>
								{{item.name}}
							</view>
							<view class="tki-tree-check" @tap.stop="_treeItemSelect(item, index)"
								v-if="(selectParent?true:item.lastRank) && item.notChecked">
								<view class="tki-tree-check-yes" v-if="item.checked" :class="{'radio':!multiple}"
									:style="{'border-color':confirmColor}">
									<view class="tki-tree-check-yes-b" :style="{'background-color':confirmColor}">
									</view>
								</view>
								<view class="tki-tree-check-no" v-else :class="{'radio':!multiple}"
									:style="{'border-color':confirmColor}"></view>
							</view>
						</view>
					</block>
				</scroll-view>
			</view>
		</view>
	</view>
</template>

<script>
	export default {
		name: "tki-tree",
		props: {
			range: {
				type: Array,
				default: function() {
					return [];
				},
			},
			idKey: {
				type: String,
				default: "id",
			},
			rangeKey: {
				type: String,
				default: "label",
			},
			title: {
				type: String,
				default: "",
			},
			multiple: {
				// 是否可以多选
				type: Boolean,
				default: false,
				// default: true
			},
			selectParent: {
				//是否可以选父级
				type: Boolean,
				default: true,
			},
			foldAll: {
				//折叠时关闭所有已经打开的子集，再次打开时需要一级一级打开
				type: Boolean,
				default: false,
			},
			confirmColor: {
				// 确定按钮颜色
				type: String,
				default: "#0194FF", // #07bb07
			},
			cancelColor: {
				// 取消按钮颜色
				type: String,
				default: "", // #757575
			},
			titleColor: {
				// 标题颜色
				type: String,
				default: "", // #757575
			},
			currentIcon: {
				// 展开时候的ic
				type: String,
				default: "",
			},
			defaultIcon: {
				// 折叠时候的ic
				type: String,
				default: "",
			},
			lastIcon: {
				// 没有子集的ic
				type: String,
				default: "",
			},
			border: {
				// 是否有分割线
				type: Boolean,
				default: false,
			},
		},
		data() {
			return {
				showTree: false,
				treeList: [],
				selectIndex: -1,
				oldTreeList: [],
				selectKey: 0,
				keyword: "",
			};
		},
		computed: {},
		methods: {
			_show(key = 0) {
				this.selectKey = key;
				this.showTree = true;
			},
			_hide() {
				this.showTree = false;
			},
			_cancel() {
				this._hide();
				this.$emit("cancel", "");
			},
			_confirm() {
				// 处理所选数据
				let rt = [],
					obj = {};
				this.treeList.forEach((v, i) => {
					if (this.treeList[i].checked) {
						obj = {};
						obj.parents = this.treeList[i].parents;
						obj = Object.assign(obj, this.treeList[i].source);
						// 移除子元素
						delete obj.children;
						rt.push(obj);
					}
				});
				this._hide();
				this.$emit("confirm", rt, this.selectKey);
			},
			//搜索
			search(val) {
				if (this.keyword) {
					let searchData = this.rebuildData(this.keyword, this.range);
					this._initTree(searchData);
				} else {
					// this.treeList = [...this.oldTreeList];
					this._initTree(this.range);
				}
			},

			rebuildData(value, arr) {
				let newarr = [];
				arr.forEach((element) => {
					if (element.children && element.children.length) {
						const ch = this.rebuildData(value, element.children);
						const obj = {
							...element,
							children: ch,
						};
						if (ch && ch.length) {
							newarr.push(obj);
						}
					} else {
						if (
							element[this.rangeKey]
							.toLowerCase()
							.indexOf(this.keyword.toLowerCase()) >= 0
						) {
							newarr.push(element);
						}
					}
				});
				return newarr;
			},
			//扁平化树结构
			_renderTreeList(list = [], rank = 0, parentId = [], parents = []) {
				list.forEach((item) => {
					this.treeList.push({
						id: item[this.idKey],
						name: item[this.rangeKey],
						source: item,
						parentId, // 父级id数组
						parents, // 父级id数组
						rank, // 层级
						showChild: rank == 0 ? true : false, //子级是否显示
						open: rank == 0 ? true : false, //是否打开
						show: rank === 0 ||  rank === 1, // 自身是否显示
						hideArr: [],
						orChecked: item.checked ? item.checked : false,
						checked: item.checked ? item.checked : false,
						notChecked: item.notChecked == undefined ||
							item.notChecked == null ||
							item.notChecked == true ?
							true :
							false, //是否展示选择框
					});
					if (Array.isArray(item.children) && item.children.length > 0) {
						// console.log(item)
						let parentid = [...parentId],
							parentArr = [...parents],
							childrenid = [...childrenid];
						delete parentArr.children;
						parentid.push(item[this.idKey]);
						parentArr.push({
							[this.idKey]: item[this.idKey],
							[this.rangeKey]: item[this.rangeKey],
						});
						this._renderTreeList(item.children, rank + 1, parentid, parentArr);
					} else {
						this.treeList[this.treeList.length - 1].lastRank = true;
					}
				});
				this.oldTreeList = [...this.treeList];
				// console.log(list)

				// console.log("this.treeList.", this.treeList);
			},
			// 处理默认选择
			_defaultSelect() {
				this.treeList.forEach((v, i) => {
					if (v.checked) {
						this.treeList.forEach((v2, i2) => {
							if (v.parentId.toString().indexOf(v2.parentId.toString()) >= 0) {
								v2.show = true;
								if (v.parentId.includes(v2.id)) {
									v2.showChild = true;
									v2.open = true;
								}
							}
						});
					}
				});
			},
			// 点击
			_treeItemTap(item, index) {
				if (item.lastRank === true) {
					//点击最后一级时触发事件
					this.treeList[index].checked = !this.treeList[index].checked;
					this._fixMultiple(index);
					return;
				}
				let list = this.treeList;
				let id = item.id;
				item.showChild = !item.showChild;
				item.open = item.showChild ? true : !item.open;
				list.forEach((childItem, i) => {
					if (item.showChild === false) {
						//隐藏所有子级
						if (!childItem.parentId.includes(id)) {
							return;
						}
						if (!this.foldAll) {
							if (childItem.lastRank !== true && !childItem.open) {
								childItem.showChild = false;
							}
							// 为隐藏的内容添加一个标记
							if (childItem.show) {
								childItem.hideArr[item.rank] = id;
							}
						} else {
							if (childItem.lastRank !== true) {
								childItem.showChild = false;
							}
						}
						childItem.show = false;
					} else {
						// 打开子集
						if (childItem.parentId[childItem.parentId.length - 1] === id) {
							childItem.show = true;
						}
						// 打开被隐藏的子集
						if (childItem.parentId.includes(id) && !this.foldAll) {
							// console.log(childItem.hideArr)
							if (childItem.hideArr[item.rank] === id) {
								childItem.show = true;
								if (childItem.open && childItem.showChild) {
									childItem.showChild = true;
								} else {
									childItem.showChild = false;
								}
								childItem.hideArr[item.rank] = null;
							}
							// console.log(childItem.hideArr)
						}
					}
				});
				// console.log(this.treeList)
			},
			_treeItemSelect(item, index) {
				this.treeList[index].checked = !this.treeList[index].checked;
				this._fixMultiple(index);
			},
			// 处理单选多选
			_fixMultiple(index) {
				if (!this.multiple) {
					// 如果是单选
					this.treeList.forEach((v, i) => {
						if (i != index) {
							this.treeList[i].checked = false;
						} else {
							this.treeList[i].checked = true;
						}
					});
				}
			},
			// 重置数据
			_reTreeList() {
				this.treeList.forEach((v, i) => {
					this.treeList[i].checked = v.orChecked;
				});
			},
			_initTree(range = this.range) {
				this.treeList = [];
				this._renderTreeList(range);
				this.$nextTick(() => {
					this._defaultSelect(range);
				});
			},
		},
		watch: {
			range(list) {
				this._initTree(list);
			},
			multiple() {
				if (this.range.length) {
					this._reTreeList();
				}
			},
			selectParent() {
				if (this.range.length) {
					this._reTreeList();
				}
			},
		},
		mounted() {
			this._initTree();
		},
	};
</script>

<style scoped>
	@import "./style.css";
</style>